# -*- coding: utf-8 -*-
from ultralytics import YOLO

if __name__ == '__main__':
    # 加载模型
    #model = YOLO("yolov8n.yaml")  # 从头开始构建新模型
    model = YOLO("runs/detect/train4/weights/best.pt")  # 加载预训练模型（建议用于训练）

    # 使用模型
    model.predict('./test_images_red_2',save=True)
    #model.predict('datasets/phoneandmouse/test/images/672368864be9e37074513e6e0acf3830.mp4',save=True)
